Goto

Collaborating Authors

 conformation generation





Chemistry-Enhanced Diffusion-Based Framework for Small-to-Large Molecular Conformation Generation

arXiv.org Artificial Intelligence

Obtaining 3D conformations of realistic polyatomic molecules at the quantum chemistry level remains challenging, and although recent machine learning advances offer promise, predicting large-molecule structures still requires substantial computational effort. Here, we introduce StoL, a diffusion model-based framework that enables rapid and knowledge-free generation of large molecular structures from small-molecule data. Remarkably, StoL assembles molecules in a LEGO-style fashion from scratch, without seeing the target molecules or any structures of comparable size during training. Given a SMILES input, it decomposes the molecule into chemically valid fragments, generates their 3D structures with a diffusion model trained on small molecules, and assembles them into diverse conformations. This fragment-based strategy eliminates the need for large-molecule training data while maintaining high scalability and transferability. By embedding chemical principles into key steps, StoL ensures faster convergence, chemically rational structures, and broad configurational coverage, as confirmed against DFT calculations.



Unified Molecule Pre-training with Flexible 2D and 3D Modalities: Single and Paired Modality Integration

arXiv.org Artificial Intelligence

Molecular representation learning plays a crucial role in advancing applications such as drug discovery and material design. Existing work leverages 2D and 3D modalities of molecular information for pre-training, aiming to capture comprehensive structural and geometric insights. However, these methods require paired 2D and 3D molecular data to train the model effectively and prevent it from collapsing into a single modality, posing limitations in scenarios where a certain modality is unavailable or computationally expensive to generate. To overcome this limitation, we propose FlexMol, a flexible molecule pre-training framework that learns unified molecular representations while supporting single-modality input. Specifically, inspired by the unified structure in vision-language models, our approach employs separate models for 2D and 3D molecular data, leverages parameter sharing to improve computational efficiency, and utilizes a decoder to generate features for the missing modality. This enables a multistage continuous learning process where both modalities contribute collaboratively during training, while ensuring robustness when only one modality is available during inference. Extensive experiments demonstrate that FlexMol achieves superior performance across a wide range of molecular property prediction tasks, and we also empirically demonstrate its effectiveness with incomplete data. Our code and data are available at https://github.com/tewiSong/FlexMol.


Predicting Molecular Conformation via Dynamic Graph Score Matching Shitong Luo

Neural Information Processing Systems

Predicting stable 3D conformations from 2D molecular graphs has been a longstanding challenge in computational chemistry. Recently, machine learning approaches have demonstrated very promising results compared to traditional experimental and physics-based simulation methods.



PolyConf: Unlocking Polymer Conformation Generation through Hierarchical Generative Models

arXiv.org Artificial Intelligence

Polymer conformation generation is a critical task that enables atomic-level studies of diverse polymer materials. While significant advances have been made in designing conformation generation methods for small molecules and proteins, these methods struggle to generate polymer conformations due to their unique structural characteristics. Meanwhile, the scarcity of polymer conformation datasets further limits the progress, making this important area largely unexplored. In this work, we propose PolyConf, a pioneering tailored polymer conformation generation method that leverages hierarchical generative models to unlock new possibilities. Specifically, we decompose the polymer conformation into a series of local conformations (i.e., the conformations of its repeating units), generating these local conformations through an autoregressive model, and then generating their orientation transformations via a diffusion model to assemble them into the complete polymer conformation. Moreover, we develop the first benchmark with a high-quality polymer conformation dataset derived from molecular dynamics simulations to boost related research in this area. The comprehensive evaluation demonstrates that PolyConf consistently outperforms existing conformation generation methods, thus facilitating advancements in polymer modeling and simulation.


MDDM: A Molecular Dynamics Diffusion Model to Predict Particle Self-Assembly

arXiv.org Artificial Intelligence

Molecular Dynamics (MD) is a powerful computational tool that lets scientists and engineers study chemical, biological, or material systems at a micro-or nano-scale. In particular, we target a materials science application of molecular self-assembly in which the goal is to model the dynamics and structure of bulk systems containing many particles that interact with one another via a specified potential energy function. By simulating the motion and interaction of particles in a molecular system, material properties can be measured from the resulting equilibrated particle structures. While MD undoubtedly provides engineers with the capacity to perform high-fidelity material simulations, it is not without its own limitations, namely computational expense. For one, very large systems (i.e. with many particles) are required to emulate the properties of a bulk material as accurately as possible.